Removing nested for loops